package com.corpdata.core.async.pool;

import java.lang.reflect.Method;
import java.util.concurrent.Executor;
import java.util.concurrent.ThreadPoolExecutor;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.aop.interceptor.AsyncUncaughtExceptionHandler;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.annotation.AsyncConfigurer;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
import com.corpdata.core.async.TaskThreadPoolConfig;
/**
 * 默认线程池配置
 * @author zealon
 *
 */
@Configuration 
public class AsyncTaskExecutePool implements AsyncConfigurer {
	
	protected final Logger logger = LoggerFactory.getLogger(this.getClass()); 
	
	@Autowired    
	private TaskThreadPoolConfig config;  //配置属性类
	
	@Override
	public Executor getAsyncExecutor() {
		ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();  
        executor.setCorePoolSize(config.getCorePoolSize());    
        executor.setMaxPoolSize(config.getMaxPoolSize());    
        executor.setQueueCapacity(config.getQueueCapacity());    
        executor.setKeepAliveSeconds(config.getKeepAliveSeconds());    
        executor.setThreadNamePrefix("taskExecutor-");    
    
        // rejection-policy：当pool已经达到max size的时候，如何处理新任务    
        // CALLER_RUNS：不在新线程中执行任务，而是由调用者所在的线程来执行    
        executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());  
        executor.initialize();    
        return executor;   
	}

	@Override
	public AsyncUncaughtExceptionHandler getAsyncUncaughtExceptionHandler() {
		return new AsyncUncaughtExceptionHandler() {  
            
            @Override  
            public void handleUncaughtException(Throwable arg0, Method arg1, Object... arg2) {  
            	logger.error("=========================="+arg0.getMessage()+"=======================", arg0);  
            	logger.error("exception method:"+arg1.getName());  
            }

        };  
	}

}
